java/Graph Editor/src/models/GraphModel.java

   1 package models;
   2
   3 import java.io.*;
   4 import java.util.List;
   5 import java.util.ArrayList;
   6 import java.util.Observable;
   7 import java.util.Observer;
   8 import java.util.Scanner;
   9
  10
  11 public class GraphModel extends Observable implements Observer{
  12         private List<GraphEdge>         edges;
  13         private List<GraphVertex>       vertexes;
  14
  15
  16         public GraphVertex[] getVertexes(){
  17                 GraphVertex[] a = new GraphVertex[vertexes.size()];
  18                 return vertexes.toArray(a);
  19         }
  20         public GraphEdge[] getEdges(){
  21                 GraphEdge[] a = new GraphEdge[edges.size()];
  22                 return edges.toArray(a);
  23         }
  24
  25         public GraphModel(){
  26                 edges = new ArrayList<GraphEdge>();
  27                 vertexes = new ArrayList<GraphVertex>();
  28         }
  29
  30         public synchronized void setChanged() {
  31                super.setChanged();
  32                notifyObservers();
  33         }
  34
  35         public void addEdge(GraphEdge ... edge){
  36                 for(int i=0; i < edge.length; i++){
  37                         edges.add(edge[i]);
  38                         edge[i].addObserver(this);
  39                 }
  40                 setChanged();
  41         }
  42
  43         public void addGraph(GraphVertex ... vertex){
  44                 for(int i=0; i < vertex.length; i++){
  45                         vertexes.add(vertex[i]);
  46                         vertex[i].addObserver(this);
  47                 }
  48                 setChanged();
  49         }
  50
  51         public void removeEdge(GraphEdge ... edge){
  52                 for(int i=0; i < edge.length; i++){
  53                         edges.remove(edge[i]);
  54                         edge[i].deleteObserver(this);
  55                 }
  56                 setChanged();
  57         }
  58         public void removeGraph(GraphVertex ... vertex){
  59                 for(int i=0; i < vertex.length; i++){
  60                         vertexes.remove(vertex[i]);
  61                         vertex[i].deleteObserver(this);
  62                 }
  63                 setChanged();
  64         }
  65
  66         public GraphVertex getGraphByCoordinate(int x, int y){
  67                 for(int i=0; i < vertexes.size(); i++){
  68                         if(vertexes.get(i).isAt(x, y))
  69                                 return vertexes.get(i);
  70                 }
  71                 return null;
  72         }
  73
  74         public void saveModel() throws IOException {
  75                 PrintWriter pw = new PrintWriter(new FileWriter("file.graph"));
  76                 pw.println(vertexes.size() + "," + edges.size());
  77
  78         }
  79
  80         @Override
  81         public void update(Observable arg0, Object arg1) {
  82                 if(vertexes.contains(arg0) || edges.contains(arg0)){
  83                         this.setChanged();
  84                 }
  85         }
  86
  87
  88         //File IO Code
  89         public void writeToFile(File file) throws Exception
  90         {
  91                 FileOutputStream fos = new FileOutputStream(file);
  92                 OutputStreamWriter out = new OutputStreamWriter(fos, "UTF-8");
  93
  94                 out.write(vertexes.size() + " " + edges.size() + "\r\n");
  95                 for(int i=0; i < vertexes.size(); i++)
  96                         out.write(vertexes.get(i).getX() + " " + vertexes.get(i).getY() + " " + vertexes.get(i).getWidth() + " " + vertexes.get(i).getHeight() +" "+ vertexes.get(i).getName() + "\r\n");
  97                 for(int i=0; i < edges.size(); i++)
  98                         out.write(vertexes.indexOf(edges.get(i).getEdge(false)) + " " + vertexes.indexOf(edges.get(i).getEdge(true)) + "\r\n");
  99                 out.flush();
 100                 out.close();
 101                 fos.flush();
 102                 fos.close();
 103         }
 104
 105
 106
 107         private static GraphVertex[] readVertexes(int count, Scanner s)
 108         {
 109                 GraphVertex[] vertexes = new GraphVertex[count];
 110                 for(int i=0; i < count; i++)
 111                         vertexes[i] = new GraphVertex(s);
 112                 return vertexes;
 113         }
 114
 115
 116         public static GraphModel fromFile(File file) throws Exception
 117         {
 118                 GraphModel model = new GraphModel();
 119                 Scanner scanner = new Scanner(file, "UTF-8");
 120
 121                 int numKnopen = scanner.nextInt();
 122                 int numKanten = scanner.nextInt();
 123                 scanner.nextLine();
 124
 125                 GraphVertex[] vertexes = readVertexes(numKnopen,scanner);
 126
 127                 model.addEdge(readKanten(scanner, numKanten, vertexes));
 128                 model.addGraph(vertexes);
 129
 130                 return model;
 131         }
 132         private static GraphEdge[] readKanten(Scanner scanner, int numKanten,
 133                 GraphVertex[] vertexes) {
 134                 GraphEdge[] edges = new GraphEdge[numKanten];
 135                 for (int i = 0; i < numKanten; i++) {
 136                         int obj1 = scanner.nextInt();
 137                         int obj2 = scanner.nextInt();
 138                         edges[i] = new GraphEdge(vertexes[obj1], vertexes[obj2]);
 139                         if(i != (numKanten-1)){
 140                                 scanner.nextLine();
 141                         }
 142                 }
 143                 return edges;
 144         }
 145
 146
 147         public static GraphModel fromFile(String fileName) throws Exception
 148         {
 149                 File file = new File(fileName);
 150                 return fromFile(file);
 151         }
 152
 153 }